/*! ******************************************************************************
 *
 * Created on 2021年11月29日
 *
 * Copyright(c) YuZhou Big Data Laboratory CO.,Ltd, 2022.  All rights reserved.
 *
 *******************************************************************************
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 ******************************************************************************/
package org.pentaho.di.ui.spoon.navigation.view.subtree;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.TreeItem;
import org.pentaho.di.core.exception.KettleException;
import org.pentaho.di.i18n.BaseMessages;
import org.pentaho.di.partition.PartitionSchema;
import org.pentaho.di.repository.ObjectId;
import org.pentaho.di.trans.TransMeta;
import org.pentaho.di.ui.core.dialog.ErrorDialog;
import org.pentaho.di.ui.core.gui.GUIResource;
import org.pentaho.di.ui.spoon.Spoon;
import org.pentaho.di.ui.spoon.navigation.view.Subtree;
import org.pentaho.di.ui.spoon.navigation.view.MetaTree;

/**
 * 数据库分区schemas子树
 * 
 * @author hs@yzbdl.ac.cn
 *
 */
public class PartitionsSubtree extends Subtree {
	private static Class<?> PKG = Spoon.class;

	/**
	 * @param parent
	 */
	public PartitionsSubtree(MetaTree parent, TransMeta meta) {
		super(parent);

		List<PartitionSchema> partitionSchemas;
		try {
			partitionSchemas = pickupPartitionSchemas(meta);
		} catch (KettleException e) {
			new ErrorDialog(this.getSpoonRef().getShell(), BaseMessages.getString(PKG, "Spoon.ErrorDialog.Title"),
					BaseMessages.getString(PKG, "Spoon.ErrorDialog.ErrorFetchingFromRepo.PartitioningSchemas"), e);
			return;
		}

		// Put the steps below it.
		for (PartitionSchema partitionSchema : partitionSchemas) {
			if (!this.getSpoonRef().filterMatch(partitionSchema.getName())) {
				continue;
			}

			TreeItem tiPartition = new TreeItem(this.getView(), SWT.NONE);
			tiPartition.setText(partitionSchema.getName());
			tiPartition.setImage(GUIResource.getInstance().getImagePartitionSchema());

			if (partitionSchema.isShared()) {
				tiPartition.setFont(GUIResource.getInstance().getFontBold());
			}
		}

	}

	private List<PartitionSchema> pickupPartitionSchemas(TransMeta transMeta) throws KettleException {
		if (this.getSpoonRef().getRep() != null) {
			ObjectId[] ids = this.getSpoonRef().getRep().getPartitionSchemaIDs(false);
			List<PartitionSchema> result = new ArrayList<>(ids.length);
			for (ObjectId id : ids) {
				PartitionSchema schema = this.getSpoonRef().getRep().loadPartitionSchema(id, null);
				result.add(schema);
			}
			return result;
		}

		return transMeta.getPartitionSchemas();
	}

	/**
	 * (non-Javadoc)
	 * 
	 * @see org.pentaho.di.ui.spoon.navigation.view.Subtree#getName()
	 */
	@Override
	public String getName() {
		return Spoon.STRING_PARTITIONS;
	}

}
